from pyecharts.charts import Bar, Timeline
from pyecharts.options import VisualMapOpts, LabelOpts, TitleOpts
from pyecharts.globals import ThemeType
f = open("D:\\file\\1960-2019全球GDP数据.csv", "r", encoding="UTF-8")
data_lines = f.readlines()
# print(data_lines)
# 关闭文件
f.close()
# 删除第一条数据
data_lines.pop(0)

# 将数据转换为字典存储, 格式为：
# {年份: [[国家, GDP],[国家, GDP]....].......}
# 先定一个字典对象
data_dict = dict()
for line in data_lines:
    year = int(line.split(",")[0])   # 年份
    country = line.split(",")[1]        # 国家
    gdp = float(line.split(",")[2])     # gdp数据
    # 如何判断字典里面有没有指定的key呢？
    try:
        data_dict[year].append([country, gdp])
    except KeyError as e:
        data_dict[year] = list()
        data_dict[year].append([country, gdp])

# 创建时间线对象
timeline = Timeline({"theme": ThemeType.LIGHT})


# print(data_dict)
sorted_year_list = sorted(data_dict.keys())
for year in sorted_year_list:
    # 对本年份的列表数据进行排序 依据元素(列表)的下标为1的元素排序
    data_dict[year].sort(key=lambda element: element[1], reverse=True)
    # 取出本年份前八名的国家数据
    year_data = data_dict[year][:8][::-1]
    x_data = list()
    y_data = list()
    for country_gdp in year_data:
        x_data.append(country_gdp[0])   # x轴存放国家
        y_data.append(country_gdp[1] / 100000000)   # y轴存放gdp数据

    # 构建柱状图
    bar = Bar()
    bar.add_xaxis(x_data)
    bar.add_yaxis("GDP(亿)", y_data, label_opts=LabelOpts(position="right"))
    # 反转X轴和Y轴
    bar.reversal_axis()
    # 设置每一年的图表标题
    bar.set_global_opts(
        title_opts=TitleOpts(title=f"{year}年全球前8GDP数据")
    )
    # 添加到时间线对象中
    timeline.add(bar, str(year))


# 设置自动播放
timeline.add_schema(
    play_interval=1000,
    is_timeline_show=True,
    is_auto_play=True,
    is_loop_play=False
)

# 绘图
timeline.render("1960-1019全球GDP前8国家.html")
